#include <boost/python.hpp>

#include "../Lattice.h"

/************* Python wrapper *************/

BOOST_PYTHON_MODULE(lattice) {
	boost::python::class_<Lattice>("Lattice", boost::python::init<>())
		.def("init", &Lattice::init)
		.def("update", &Lattice::update)
		.def("getParticlePosition", &Lattice::getParticlePosition,
				boost::python::return_value_policy<boost::python::reference_existing_object>())
		.def("getNParticles", &Lattice::getNParticles)
		.def("setU", &Lattice::setU)
		.def("resetGrid", &Lattice::resetGrid)
	;	
	boost::python::class_<vec<double> >("VecDouble", boost::python::init<double, double >())
		.def("norm", &vec<double>::norm)
	;
}
